package cn.bjxy.shishaolong.utils;

import cn.bjxy.shishaolong.common.StatusCode;
import cn.bjxy.shishaolong.system.exception.OperatorFailureException;
import io.jsonwebtoken.*;
import io.jsonwebtoken.security.Keys;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.PropertySource;
import org.springframework.stereotype.Component;

import java.security.KeyPair;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;

/**
 * token工具类
 *
 * @Author 时少龙
 * @Date 2019-03-24 19:41
 * @Version 1.0
 */
@Component
@PropertySource("classpath:config/system-config.properties")
public class TokenUtil {

    //    @Value("${token.secret}")
//    private String tokenSecret;
    @Value("${token.expire}")
    private String defaultTokenExpire;
    private KeyPair keyPair;

    /**
     * token中存放userId的键
     */
    public static final String TOKEN_USER_ID = "userId";
    private static final String TOKEN_SUBJECT = "时少龙";

    {
        // 初始化秘钥
        keyPair = Keys.keyPairFor(SignatureAlgorithm.RS256);
    }

    /**
     * 生成token
     *
     * @param subject         token的所属者
     * @param claims          token的payload  token中存放的信息
     * @param tokenExpireTime 当前时间戳 传入小于0的数使用默认过期时间
     * @return token
     */
    public String createToken(long tokenExpireTime, String subject, Map claims) {
        JwtBuilder jwtBuilder = Jwts.builder().setSubject(subject).setIssuedAt(new Date())
                .setClaims(claims);
        //设置过期时间
        long expireTime = System.currentTimeMillis() + Long.parseLong(defaultTokenExpire);
        if (tokenExpireTime > 0) {
            expireTime = tokenExpireTime;
        }
        jwtBuilder.setExpiration(new Date(expireTime));
        // 签名
        jwtBuilder.signWith(keyPair.getPrivate());
        return jwtBuilder.compact();
    }

    /**
     * 使用默认值 生成token
     *
     * @param userId 用户id
     * @return token
     */
    public String createToken(Integer userId) {
        Map<String, Object> claims = new HashMap<>();
        claims.put(TOKEN_USER_ID, userId);
        return createToken(-1, TOKEN_SUBJECT, claims);
    }

    /**
     * 检测token
     *
     * @param token
     * @return
     */
    public boolean checkToken(String token) {
        if (parseToken(token) != null) {
            return true;
        }
        return false;
    }

    /**
     * 获取subject
     *
     * @param token
     * @return
     */
    public Integer getUserId(String token) {
        return (Integer) getClaim(token).get(TOKEN_USER_ID);
    }

    /**
     * 获取token中包含的信息
     *
     * @param token
     * @return
     */
    public Claims getClaim(String token) {
        return parseToken(token).getBody();
    }

    private Jwt<Header, Claims> parseToken(String token) {
        try {
            return Jwts.parser().setSigningKey(keyPair.getPublic()).parse(token);
        } catch (JwtException ex) {
            ex.printStackTrace();
            throw new OperatorFailureException(StatusCode.TOKEN_AUTHEN_FAILURE, StatusCode.TOKEN_AUTHEN_FAILURE_MSG);
        }
    }

}
